package com.puff.framework.utils;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.geom.QuadCurve2D;
import java.awt.image.BufferedImage;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

import javax.imageio.ImageIO;

/**
 * 生产验证码
 * 
 * @author 董超
 * @date 2013-3-19 下午2:44:58
 */
public class AuthCodeUtil {
	private final static Random random = new Random();
	// 随机字体样式
	private final static int[] fontStyle = { Font.HANGING_BASELINE, Font.ITALIC, Font.LAYOUT_LEFT_TO_RIGHT, Font.LAYOUT_NO_LIMIT_CONTEXT, Font.LAYOUT_NO_START_CONTEXT,
			Font.LAYOUT_RIGHT_TO_LEFT, Font.CENTER_BASELINE, Font.PLAIN, Font.ROMAN_BASELINE, Font.TRUETYPE_FONT, Font.TYPE1_FONT, Font.BOLD };

	private final static String[] font = { "Ravie", "Forte", "Arial", "Courier" };

	private final static int[] xarr = { 23, 24, 19, 20, 21 };
	private final static int[] yarr = { 20, 21, 22, 23, 24, 25, 16, 17, 18 };
	private final static int[] font_size = { 28, 29, 30, 21, 22, 23, 24, 27, 26, 25 };

	/**
	 * 画随机码图
	 * 
	 * @param out
	 * @param width
	 * @param height
	 * @throws IOException
	 */
	public static void draw(OutputStream out, String value) throws IOException {
		int width = 100, height = 40;
		BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		Graphics2D g = (Graphics2D) bi.getGraphics();
		g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
		g.setColor(randColor(240, 255));
		g.fillRect(0, 0, width, height);
		g.drawRect(1, 1, width - 2, height - 2);
		for (int i = 0; i < 10; i++) {
			g.setColor(randColor(150, 250));
			g.drawOval(random.nextInt(110), random.nextInt(24), 5 + random.nextInt(10), 5 + random.nextInt(10));
		}
		g.setFont(getFont());
		g.setColor(randColor(xarr[(int) (Math.random() * 5)], 254));
		for (int i = 0, len = value.length(); i < len; i++) {
			String rand = String.valueOf(value.charAt(i));
			int degree = random.nextInt(23);
			if (i % 2 == 0) {
				degree = degree * (-1);
			}
			int x = xarr[(int) (Math.random() * 5)] * i, y = yarr[(int) (Math.random() * 8)];
			g.rotate(Math.toRadians(degree), x, y);
			g.setColor(randColor(48, 254));
			g.drawString(rand, x + 8, y + 10);
			g.rotate(-Math.toRadians(degree), x, y);
		}
		// 图片中间线
		g.setColor(randColor(0, 200));
		// width是线宽,float型
		BasicStroke bs = new BasicStroke(2);
		g.setStroke(bs);
		// 画出曲线
		QuadCurve2D.Double curve = new QuadCurve2D.Double(0d, random.nextInt(height - 8) + 4, width / 2, height / 2, width, random.nextInt(height - 8) + 4);
		g.draw(curve);
		// 销毁图像
		g.dispose();
		ImageIO.write(bi, "png", out);
	}

	private static Font getFont() {
		return new Font(font[(int) (Math.random() * 4)], fontStyle[(int) (Math.random() * 12)], font_size[(int) (Math.random() * 10)]);
	}

	private static Color randColor(int fc, int bc) {// 给定范围获得随机颜色
		if (fc > 255) {
			fc = 255;
		}
		if (bc > 255) {
			bc = 255;
		}
		int r = fc + random.nextInt(bc - fc);
		int g = fc + random.nextInt(bc - fc);
		int b = fc + random.nextInt(bc - fc);
		return new Color(r, g, b);
	}

	public static void main(String[] args) throws IOException {
		FileOutputStream out = new FileOutputStream("d:\\aa.png");
		draw(out, getRandom(4));
	}

	private static char[] c = { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j', 'k', 'm', 'n', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', 'A', 'B', 'C', 'D', 'E', 'F',
			'G', 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '1', '3', '5', '6', '7', '8', '9' };

	public static String getRandom(int size) {
		StringBuilder sb = new StringBuilder(size);
		for (int i = 0; i < size; i++) {
			sb.append(c[Math.abs(random.nextInt()) % c.length]);
		}
		return sb.toString();
	}

}
